package com.ruoyi.system.service.impl;

import java.math.BigDecimal;
import java.util.Collection;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.system.domain.Dish;
import com.ruoyi.system.domain.dto.ReportMaterialDTO;
import com.ruoyi.system.domain.vo.ReportDishMaterialVO;
import com.ruoyi.system.domain.vo.ReportMaterialVO;
import com.ruoyi.system.mapper.DishMapper;
import com.ruoyi.system.mapper.DishMaterialMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.ruoyi.system.mapper.WeddingBanquetMapper;
import com.ruoyi.system.domain.WeddingBanquet;
import com.ruoyi.system.service.IWeddingBanquetService;

/**
 * 【请填写功能名称】Service业务层处理
 *
 * @author ruoyi
 * @date 2025-03-21
 */
@Service
public class WeddingBanquetServiceImpl implements IWeddingBanquetService {
    @Autowired
    private WeddingBanquetMapper weddingBanquetMapper;

    @Autowired
    private DishMaterialMapper dishMaterialMapper;

    /**
     * 查询【请填写功能名称】
     *
     * @param id 【请填写功能名称】主键
     * @return 【请填写功能名称】
     */
    @Override
    public WeddingBanquet selectWeddingBanquetById(String id) {
        return weddingBanquetMapper.selectWeddingBanquetById(id);
    }

    /**
     * 查询【请填写功能名称】列表
     *
     * @param weddingBanquet 【请填写功能名称】
     * @return 【请填写功能名称】
     */
    @Override
    public List<WeddingBanquet> selectWeddingBanquetList(WeddingBanquet weddingBanquet) {
        return weddingBanquetMapper.selectWeddingBanquetList(weddingBanquet);
    }

    /**
     * 新增【请填写功能名称】
     *
     * @param weddingBanquet 【请填写功能名称】
     * @return 结果
     */
    @Override
    public int insertWeddingBanquet(WeddingBanquet weddingBanquet) {
        // 获取用户名
        String username = SecurityUtils.getUsername();
        weddingBanquet.setCreatedBy(username);
        weddingBanquet.setUpdatedBy(username);
        return weddingBanquetMapper.insertWeddingBanquet(weddingBanquet);
    }

    /**
     * 修改【请填写功能名称】
     *
     * @param weddingBanquet 【请填写功能名称】
     * @return 结果
     */
    @Override
    public int updateWeddingBanquet(WeddingBanquet weddingBanquet) {
        return weddingBanquetMapper.updateWeddingBanquet(weddingBanquet);
    }

    /**
     * 批量删除【请填写功能名称】
     *
     * @param ids 需要删除的【请填写功能名称】主键
     * @return 结果
     */
    @Override
    public int deleteWeddingBanquetByIds(String[] ids) {
        return weddingBanquetMapper.deleteWeddingBanquetByIds(ids);
    }

    /**
     * 删除【请填写功能名称】信息
     *
     * @param id 【请填写功能名称】主键
     * @return 结果
     */
    @Override
    public int deleteWeddingBanquetById(String id) {
        return weddingBanquetMapper.deleteWeddingBanquetById(id);
    }

    @Override
    public List<ReportMaterialVO> reportMaterial(ReportMaterialDTO dto) {
        Integer banquetId = dto.getBanquetId();
        WeddingBanquet weddingBanquet = this.selectWeddingBanquetById(String.valueOf(banquetId));
        Long tableCount = weddingBanquet.getTableCount();

        List<Integer> dishIds = dto.getDishIds();
        List<ReportDishMaterialVO> list = dishMaterialMapper.selectDishMaterialListByDishIds(dishIds);
        List<ReportMaterialVO> voList = list.stream().flatMap(e -> e.getList().stream())
                .collect(Collectors.toList()).stream()
                // 按materialId分组
                .collect(Collectors.groupingBy(ReportMaterialVO::getMaterialId))
                .values().stream()
                .map(group -> {
                    // 计算perTable的总和
                    BigDecimal sum = group.stream()
                            .map(ReportMaterialVO::getPerTable)
                            .reduce(BigDecimal.ZERO, BigDecimal::add);
                    BigDecimal total = sum.multiply(new BigDecimal(tableCount));
                    // 取分组中的第一个元素获取其他字段
                    ReportMaterialVO first = group.get(0);
                    ReportMaterialVO reportMaterialVO = new ReportMaterialVO();
                    reportMaterialVO.setMaterialId(first.getMaterialId());
                    reportMaterialVO.setMaterialName(first.getMaterialName());
                    reportMaterialVO.setUnit(first.getUnit());
                    reportMaterialVO.setCategory(first.getCategory());
                    reportMaterialVO.setPerTable(total);
                    return reportMaterialVO;
                })
                .sorted(Comparator.comparing(ReportMaterialVO::getCategory))
                .collect(Collectors.toList());
        return voList;
    }
}
